Prime Number Program In Python Using Function